from trafficlights import TrafficLights